iT邦幫忙

2022 iThome 鐵人賽

DAY 5
0
AI & Data

【30天之新手學習筆記】PyTorch系列 第 5

Day 05 - 將圖片資料利用張量表示

  • 分享至 

  • xImage
  •  

今天的筆記要介紹如何利用imageio模組來載入jpg檔的圖片.範例中會使用預先安裝好PyTorch模組的Jupyter來進行範例的實際操作,在進行操作前要先準備一張圖片以便進行後續的操作.這個操作的運行,我們會使用到三個模組分別為torch,imageio與numpy,最後利用permute()函式來對張量進行調整.

以張量來表示圖片

  1. 一開始我們必須在Jupyter上安裝imageio模組
    https://ithelp.ithome.com.tw/upload/images/20220918/20152671gWzs7EfAdW.jpg
  2. 載入將圖片轉化為張量需用到的三個模組(torch,imageio,numpy)
    https://ithelp.ithome.com.tw/upload/images/20220918/20152671OmbAwmYJF9.jpg
  3. 接下來我們利用imageio將圖片載入,我們必須利用圖片的路徑來完成,要注意的是這裡的路徑一層一層的表達方式並不是我們在本機端常用的 \ 而是 / .
    例如 : '\Users\Golden\Desktop\cat_like_a_tiger.jpg'要順利的利用imageio來讀取必須要改成'/Users/Golden/Desktop/cat_like_a_tiger.jpg'
    https://ithelp.ithome.com.tw/upload/images/20220918/20152671rc3DZ7gqkT.jpg
    img_arr是一個類似於Numpy陣列的三軸物件,三軸分別為圖片的高度,寬度以及色彩通道(H X W X C)
  4. 由於PyTorch要求圖片資料張量必須以(C X H X W)為軸的順序,因此我們要利用permute()函式來調整
    https://ithelp.ithome.com.tw/upload/images/20220918/20152671D9BVDkLFnr.jpg

透過上述四個簡單的步驟,我們就可以將我們要帶入PyTorch中進行操作的圖片資料張量做初步的處理.
今天的學習筆記是我這個初學者在學習的過程中探索到的一些小重點,當然在AI大神面前這些都不足為奇,但我還是會持續每天的發文的XD.


參考資料:


上一篇
Day 04 - 張量(Tensor)的介紹與操作
下一篇
Day 06 - 建立一個四軸張量儲存多張圖片資訊
系列文
【30天之新手學習筆記】PyTorch30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言